Description

Motivation:
when adding optional model families, in current setting, LLM_MODELS value varies based on import result

try:
    from langchain_anthropic import ChatAnthropic
except ImportError:
    ChatAnthropic=None

if ChatAnthropic in not None:
    LLM_MODELS = Union[ChatOpenAI, ChatAnthropic]

Use the parent class of all langchain supported chat models
Then we don't need this lines of code:

# TODO add new wrappers in if new model support is added
LLM_MODELS = Union[ChatOpenAI]

if ChatAnthropic in not None:
    LLM_MODELS = Union[ChatOpenAI, ChatAnthropic]

Screenshot

Notice

• I acknowledge and agree that, by checking this box and clicking "Submit Pull Request":

  • I submit this contribution under the Apache 2.0 license and represent that I am entitled to do so on behalf of myself, my employer, or relevant third parties, as applicable.
  • I certify that (a) this contribution is my original creation and / or (b) to the extent it is not my original creation, I am authorized to submit this contribution on behalf of the original creator(s) or their licensees.
  • I certify that the use of this contribution as authorized by the Apache 2.0 license does not violate the intellectual property rights of anyone else.
  • I have not referenced individuals, products or companies in any commits, directly or indirectly.
  • I have not added data or restricted code in any commits, directly or indirectly.